LimesurveyVariableEntityProvider.java

/*
 * Copyright (c) 2017 OBiBa. All rights reserved.
 *
 * This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.obiba.magma.datasource.limesurvey;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Collections;
import java.util.List;
import java.util.Set;

import javax.validation.constraints.NotNull;

import org.obiba.magma.Datasource;
import org.obiba.magma.Initialisable;
import org.obiba.magma.VariableEntity;
import org.obiba.magma.support.AbstractVariableEntityProvider;
import org.obiba.magma.support.VariableEntityBean;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.BadSqlGrammarException;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;

import com.google.common.collect.Lists;
import com.google.common.collect.Sets;

public class LimesurveyVariableEntityProvider extends AbstractVariableEntityProvider implements Initialisable {

  private static final Logger log = LoggerFactory.getLogger(LimesurveyVariableEntityProvider.class);

  private Set<VariableEntity> entities;

  private final Integer sid;

  private final LimesurveyDatasource datasource;

  protected LimesurveyVariableEntityProvider(String entityType, Datasource datasource, Integer sid) {
    super(entityType);
    this.datasource = (LimesurveyDatasource) datasource;
    this.sid = sid;
  }

  @Override
  public void initialise() {
    String sqlEntities = "SELECT " + datasource.quoteAndPrefix("token") + " FROM " +
        datasource.quoteAndPrefix("survey_" + sid) + " WHERE " + datasource.quoteAndPrefix("submitdate") +
        " is not NULL and " + datasource.quoteAndPrefix("token") + " is not NULL";
    JdbcTemplate jdbcTemplate = new JdbcTemplate(datasource.getDataSource());

    List<VariableEntity> entityList = null;
    try {
      entityList = jdbcTemplate.query(sqlEntities, new RowMapper<VariableEntity>() {
        @Override
        public VariableEntity mapRow(ResultSet rs, int rowNum) throws SQLException {
          String entityId = rs.getString("token");
          return new VariableEntityBean(LimesurveyValueTable.PARTICIPANT, entityId);
        }
      });
    } catch(BadSqlGrammarException e) {
      entityList = Lists.newArrayList();
      log.info("survey_{} is probably not active", sid);
    }
    entities = Sets.newHashSet(entityList);
  }

  @NotNull
  @Override
  public Set<VariableEntity> getVariableEntities() {
    return Collections.unmodifiableSet(entities);
  }
}